state_dict_keys = {
    "diffusion_model.double_blocks.0.img_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.0.img_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.0.img_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.0.img_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.0.img_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.0.img_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.0.img_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.0.img_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.0.txt_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.0.txt_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.0.txt_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.0.txt_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.0.txt_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.0.txt_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.0.txt_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.0.txt_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.1.img_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.1.img_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.1.img_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.1.img_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.1.img_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.1.img_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.1.img_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.1.img_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.1.txt_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.1.txt_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.1.txt_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.1.txt_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.1.txt_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.1.txt_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.1.txt_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.1.txt_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.10.img_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.10.img_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.10.img_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.10.img_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.10.img_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.10.img_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.10.img_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.10.img_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.10.txt_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.10.txt_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.10.txt_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.10.txt_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.10.txt_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.10.txt_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.10.txt_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.10.txt_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.11.img_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.11.img_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.11.img_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.11.img_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.11.img_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.11.img_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.11.img_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.11.img_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.11.txt_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.11.txt_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.11.txt_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.11.txt_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.11.txt_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.11.txt_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.11.txt_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.11.txt_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.12.img_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.12.img_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.12.img_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.12.img_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.12.img_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.12.img_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.12.img_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.12.img_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.12.txt_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.12.txt_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.12.txt_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.12.txt_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.12.txt_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.12.txt_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.12.txt_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.12.txt_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.13.img_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.13.img_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.13.img_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.13.img_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.13.img_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.13.img_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.13.img_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.13.img_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.13.txt_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.13.txt_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.13.txt_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.13.txt_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.13.txt_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.13.txt_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.13.txt_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.13.txt_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.14.img_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.14.img_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.14.img_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.14.img_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.14.img_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.14.img_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.14.img_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.14.img_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.14.txt_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.14.txt_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.14.txt_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.14.txt_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.14.txt_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.14.txt_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.14.txt_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.14.txt_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.15.img_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.15.img_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.15.img_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.15.img_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.15.img_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.15.img_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.15.img_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.15.img_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.15.txt_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.15.txt_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.15.txt_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.15.txt_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.15.txt_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.15.txt_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.15.txt_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.15.txt_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.16.img_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.16.img_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.16.img_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.16.img_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.16.img_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.16.img_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.16.img_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.16.img_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.16.txt_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.16.txt_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.16.txt_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.16.txt_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.16.txt_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.16.txt_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.16.txt_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.16.txt_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.17.img_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.17.img_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.17.img_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.17.img_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.17.img_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.17.img_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.17.img_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.17.img_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.17.txt_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.17.txt_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.17.txt_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.17.txt_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.17.txt_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.17.txt_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.17.txt_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.17.txt_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.18.img_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.18.img_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.18.img_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.18.img_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.18.img_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.18.img_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.18.img_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.18.img_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.18.txt_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.18.txt_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.18.txt_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.18.txt_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.18.txt_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.18.txt_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.18.txt_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.18.txt_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.2.img_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.2.img_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.2.img_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.2.img_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.2.img_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.2.img_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.2.img_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.2.img_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.2.txt_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.2.txt_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.2.txt_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.2.txt_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.2.txt_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.2.txt_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.2.txt_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.2.txt_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.3.img_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.3.img_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.3.img_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.3.img_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.3.img_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.3.img_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.3.img_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.3.img_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.3.txt_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.3.txt_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.3.txt_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.3.txt_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.3.txt_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.3.txt_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.3.txt_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.3.txt_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.4.img_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.4.img_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.4.img_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.4.img_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.4.img_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.4.img_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.4.img_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.4.img_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.4.txt_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.4.txt_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.4.txt_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.4.txt_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.4.txt_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.4.txt_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.4.txt_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.4.txt_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.5.img_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.5.img_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.5.img_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.5.img_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.5.img_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.5.img_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.5.img_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.5.img_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.5.txt_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.5.txt_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.5.txt_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.5.txt_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.5.txt_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.5.txt_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.5.txt_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.5.txt_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.6.img_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.6.img_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.6.img_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.6.img_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.6.img_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.6.img_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.6.img_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.6.img_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.6.txt_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.6.txt_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.6.txt_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.6.txt_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.6.txt_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.6.txt_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.6.txt_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.6.txt_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.7.img_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.7.img_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.7.img_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.7.img_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.7.img_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.7.img_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.7.img_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.7.img_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.7.txt_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.7.txt_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.7.txt_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.7.txt_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.7.txt_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.7.txt_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.7.txt_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.7.txt_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.8.img_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.8.img_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.8.img_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.8.img_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.8.img_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.8.img_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.8.img_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.8.img_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.8.txt_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.8.txt_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.8.txt_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.8.txt_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.8.txt_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.8.txt_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.8.txt_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.8.txt_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.9.img_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.9.img_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.9.img_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.9.img_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.9.img_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.9.img_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.9.img_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.9.img_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.9.txt_attn.proj.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.9.txt_attn.proj.lora_B.weight": [3072, 16],
    "diffusion_model.double_blocks.9.txt_attn.qkv.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.9.txt_attn.qkv.lora_B.weight": [9216, 16],
    "diffusion_model.double_blocks.9.txt_mlp.0.lora_A.weight": [16, 3072],
    "diffusion_model.double_blocks.9.txt_mlp.0.lora_B.weight": [12288, 16],
    "diffusion_model.double_blocks.9.txt_mlp.2.lora_A.weight": [16, 12288],
    "diffusion_model.double_blocks.9.txt_mlp.2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.0.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.0.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.0.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.0.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.1.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.1.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.1.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.1.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.10.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.10.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.10.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.10.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.11.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.11.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.11.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.11.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.12.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.12.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.12.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.12.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.13.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.13.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.13.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.13.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.14.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.14.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.14.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.14.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.15.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.15.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.15.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.15.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.16.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.16.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.16.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.16.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.17.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.17.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.17.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.17.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.18.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.18.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.18.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.18.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.19.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.19.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.19.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.19.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.2.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.2.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.2.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.2.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.20.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.20.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.20.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.20.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.21.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.21.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.21.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.21.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.22.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.22.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.22.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.22.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.23.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.23.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.23.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.23.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.24.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.24.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.24.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.24.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.25.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.25.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.25.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.25.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.26.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.26.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.26.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.26.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.27.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.27.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.27.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.27.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.28.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.28.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.28.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.28.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.29.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.29.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.29.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.29.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.3.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.3.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.3.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.3.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.30.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.30.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.30.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.30.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.31.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.31.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.31.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.31.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.32.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.32.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.32.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.32.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.33.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.33.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.33.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.33.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.34.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.34.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.34.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.34.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.35.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.35.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.35.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.35.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.36.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.36.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.36.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.36.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.37.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.37.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.37.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.37.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.4.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.4.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.4.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.4.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.5.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.5.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.5.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.5.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.6.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.6.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.6.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.6.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.7.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.7.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.7.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.7.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.8.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.8.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.8.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.8.linear2.lora_B.weight": [3072, 16],
    "diffusion_model.single_blocks.9.linear1.lora_A.weight": [16, 3072],
    "diffusion_model.single_blocks.9.linear1.lora_B.weight": [21504, 16],
    "diffusion_model.single_blocks.9.linear2.lora_A.weight": [16, 15360],
    "diffusion_model.single_blocks.9.linear2.lora_B.weight": [3072, 16],
}
